from django.conf import settings
from django.db import models
from django.contrib.auth.models import AbstractUser
from itsdangerous import TimedJSONWebSignatureSerializer as TJSSerializer, BadData

from shopping_mall.utils.models import BaseModel
from . import constants

# Create your models here.



class User(AbstractUser):
    """用户模型类"""
    mobile = models.CharField(max_length=11, unique=True, verbose_name='手机号')
    # 记录邮件的状态（是否被激活）
    email_active = models.BooleanField(default=False, verbose_name='邮箱验证状态')
    # 创建默认地址
    default_address = models.ForeignKey('Address', related_name='users', null=True, blank=True,on_delete=models.SET_NULL, verbose_name='默认地址')
    class Meta:
        db_table = 'tb_users'
        verbose_name = '用户'
        verbose_name_plural = verbose_name


    def generate_email_verify_url(self):
        """
       生成邮箱验证时的激活链接
       :return: 激活链接
       提示：self，就是该实例方法的调用者，此处是当前的更新邮件的登录用户
       """
        # 创建序列化器对象
        serializer = TJSSerializer(settings.SECRET_KEY,expires_in=constants.VERIFY_EMAIL_TOKEN_EXPIRES)
        # 准备原始数据
        data = {
            'user_id':self.id,
            'email':self.email
        }
        # 进行签名
        token = serializer.dumps(data).decode()

        # 使用token拼接完整的激活链接
        verify_url = 'http://www.meiduo.site:8080/success_verify_email.html?token=' + token
        return verify_url

    @staticmethod  #定义为静态方法， 可以直接用类来调用，要不然直接调用会产生冲突
    def check_email_verify_token(token):
        """
        验证token并提取用户信息
        :param token: 签名后的token
        :return: user  None
        """
        # 创建序列化器
        serializer = TJSSerializer(settings.SECRET_KEY,expires_in=constants.VERIFY_EMAIL_TOKEN_EXPIRES)

        try:
            # 解出签名后的token
            data = serializer.loads(token)
        except BadData:
            return None
        else:
            user_id = data.get('user_id')
            email = data.get('email')
            # 使用user_id 和email作为条件,出现出user
            try:
                return User.objects.get(id = user_id,email = email)
            except User.DoesNotExist:
                return None
            else:
                return user


class Address(BaseModel):
    """用户地址"""
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='addresses', verbose_name='用户')
    title = models.CharField(max_length=20, verbose_name='地址名称')
    receiver = models.CharField(max_length=20, verbose_name='收货人')
    province = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='province_addresses', verbose_name='省')
    city = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='city_addresses', verbose_name='市')
    district = models.ForeignKey('areas.Area', on_delete=models.PROTECT, related_name='district_addresses', verbose_name='区')
    place = models.CharField(max_length=50, verbose_name='地址')
    mobile = models.CharField(max_length=11, verbose_name='手机')
    tel = models.CharField(max_length=20, null=True, blank=True, default='', verbose_name='固定电话')
    email = models.CharField(max_length=30, null=True, blank=True, default='', verbose_name='电子邮箱')
    is_deleted = models.BooleanField(default=False, verbose_name='逻辑删除')

    class Meta:
        db_table = 'tb_address'
        verbose_name = '用户地址'
        verbose_name_plural = verbose_name
        ordering = ['-update_time']
